package com.github.yangyishe.p100;

import com.github.yangyishe.ListNode;

/**
 * 61. 旋转链表
 * https://leetcode.cn/problems/rotate-list/?envType=study-plan-v2&envId=top-interview-150
 *
 * 给你一个链表的头节点 head ，旋转链表，将链表每个节点向右移动 k 个位置。
 *
 *
 *
 * 示例 1：
 *
 *
 * 输入：head = [1,2,3,4,5], k = 2
 * 输出：[4,5,1,2,3]
 * 示例 2：
 *
 *
 * 输入：head = [0,1,2], k = 4
 * 输出：[2,0,1]
 *
 *
 * 提示：
 *
 * 链表中节点的数目在范围 [0, 500] 内
 * -100 <= Node.val <= 100
 * 0 <= k <= 2 * 109
 */
public class Problem61 {
    public static void main(String[] args) {
        int[] arr=new int[]{1,2,3,4,5};
        ListNode head = ListNode.getInstance(arr);
        int k=2;

        Problem61 problem61 = new Problem61();
        ListNode result = problem61.rotateRight(head, k);
        result.print();
    }

    /**
     * 思路:
     * 本题所要的结果, 是将链表首尾相连,
     * 同时将尾部从向左数第k%n个元素(数学含义)与其左侧的元素断开. 等同于, 将下标(size-minOffset-1)个元素的next, 设置为null
     *
     * 1. 遍历head, 获取size, 以及尾部元素
     * 2. 使用k%size, 获取实际的最小偏移量minOffset
     * 3. 判断如果minOffset==0, 则不作任何处理, 直接返回head. 否则将尾部元素next设置为head, 并遍历到index=size-minOffset-1的元素上, 记录其next元素, 并断开连接, 返回原始next元素
     *
     *
     * @param head
     * @param k
     * @return
     */
    public ListNode rotateRight(ListNode head, int k) {
        if(head==null){
            return null;
        }
        ListNode tmp=head;
        ListNode tail=null;
        int size=0;
        while(tmp!=null){
            size++;
            if(tmp.next==null){
                tail=tmp;
            }
            tmp=tmp.next;
        }

        int minOffset=k%size;

        if(minOffset==0){
            return head;
        }
        ListNode temp=head;
        for(int i=0;i<size-minOffset-1;i++){
            temp=temp.next;
        }
        ListNode newResult=temp.next;
        temp.next=null;
        tail.next=head;


        return newResult;
    }
}
